package com.example.mcp.client;

import java.util.Map;

/**
 * @author ssk
 * @date 2025/8/8
 **/
public class McpWebSocketExample {

    public static void main(String[] args) {
        McpWebSocketClient wsClient = new McpWebSocketClient("ws://localhost:8080/mcp");

        try {
            // 1. 连接
            wsClient.connect();

            // 2. 初始化
            wsClient.sendInitialize();

            // 3. 调用工具
            // 计算工具
            wsClient.callTool("calculate", Map.of("expression", "2 + 2"));

            // AI 助手
            wsClient.callTool("ai_assistant", Map.of("query", "Spring AI 是什么？"));

            // 等待响应（实际应用中应该有更好的响应处理机制）
            Thread.sleep(300000);

            // 4. 断开连接
            wsClient.disconnect();

        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}
